using python find index value of number in list